package TreeNodeSecond.SwordOffer;

public class PruneTree {
    /**
     * 剑指 Offer II 047. 二叉树剪枝
     * 给定一个二叉树 根节点 root ，树的每个节点的值要么是 0，要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。
     * <p>
     * 节点 node 的子树为 node 本身，以及所有 node 的后代。
     * <p>
     * 示例 1:
     * <p>
     * 输入: [1,null,0,0,1]
     * 输出: [1,null,0,null,1]
     * 解释:
     * 只有红色节点满足条件“所有不包含 1 的子树”。
     * 右图为返回的答案。
     *
     * @param root
     * @return
     */
    public TreeNode pruneTree(TreeNode root) {
        if (root == null) {
            return root;
        }
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);
        if (root.val == 0 && root.left == null && root.right == null) {
            root = null;
        }
        return root;
    }
}
